package com.navinfo.tripanalysis.offline.pojo;

import lombok.Data;
import lombok.ToString;

import java.io.Serializable;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 分批加载数据对应的参数
 * @author 沈东生
 */
@Data
public class LoadDataParam implements Serializable {
    /**
     *加载的日期，格式yyyyMMdd
     */
    private Date day;
    /**
     * 加载特定的tid列表
     */
    private List<Long> tidList;
    /**
     * 按批加载数据时的hashtid列表
     */
    private List<Integer> batchList;
    /**
     * 扩展参数<br/>
     */
    private Map<String, Object> extendParams = new HashMap<>();

    /**
     * 获取指定key的扩展参数<br/>
     * batchList为分批加载的tid列表，最多32个分批
     * @param key
     * @return
     */
    public Object getExtendParam(String key) {
        return extendParams.get(key);
    }

    /**
     * 设置指定key扩展参数
     * @param key
     * @param subList
     */
    public void putExtendParam(String key, List<Integer> subList) {
        extendParams.put(key, subList);
    }
}
